How to group a json by a nested key using Python